# example6_7.py
# -*- coding: utf-8 -*-
import numpy as np
import matplotlib.pyplot as plt

open_price, high_price, low_price, close_price = np.loadtxt(
    "stock.csv", delimiter=",", usecols=(1, 2, 3, 4), unpack=True, skiprows=1
)

x = np.arange(len(open_price))
plt.rcParams["font.sans-serif"] = ["SimHei"]
# 创建图像
plt.figure(figsize=(11, 7))  # 设置图像大小

# 方式1
# 创建子图
# 第1行第1列的子图
ax1 = plt.subplot(2, 2, 1)
# 第1行第2列的子图
ax2 = plt.subplot(2, 2, 2)
# 第2行第1列的子图
ax3 = plt.subplot(2, 2, 3)
# 第2行第2列的子图
ax4 = plt.subplot(2, 2, 4)
# 选择子图ax1
plt.sca(ax1)
plt.plot(x, open_price, color="red")
# 设置刻度字体大小
plt.xticks(fontsize=15)
plt.yticks(fontsize=15)
plt.title("开盘价", fontsize=15)

# 选择子图ax2
plt.sca(ax2)
plt.plot(x, close_price, "b--")
# 设置刻度字体大小
plt.xticks(fontsize=15)
plt.yticks(fontsize=15)
plt.title("收盘价", fontsize=15)

# 选择子图ax3
plt.sca(ax3)
plt.plot(x, high_price, "g--")
# 设置刻度字体大小
plt.xticks(fontsize=15)
plt.yticks(fontsize=15)
plt.title("最高价", fontsize=15)

# 选择子图ax4
plt.sca(ax4)
plt.plot(x, low_price)
# 设置刻度字体大小
plt.xticks(fontsize=15)
plt.yticks(fontsize=15)
plt.title("最低价", fontsize=15)

"""
#方式2
plt.rcParams['font.size']=15  #设置全局默认字体大小
# 子图1
plt.subplot(2,2,1) 
plt.plot(x,open_price,color='red')
# 设置刻度和标题
plt.xticks()
plt.yticks()
plt.title('开盘价')

# 子图2
plt.subplot(2,2,2) 
plt.plot(x,close_price,'b--')
# 设置刻度和标题
plt.xticks()
plt.yticks()
plt.title('收盘价')

# 子图3
plt.subplot(2,2,3) 
plt.plot(x,high_price,'g--')
# 设置刻度和标题
plt.xticks()
plt.yticks()
plt.title('最高价')

# 子图4
plt.subplot(2,2,4) 
plt.plot(x,low_price)
# 设置刻度和标题
plt.xticks()
plt.yticks()
plt.title('最低价')
"""

# 调整子图间距
plt.subplots_adjust(wspace=0.2, hspace=0.5)
plt.show()
